import com.fasterxml.jackson.databind.ObjectMapper;
import com.mysql.jdbc.jdbc2.optional.MysqlDataSource;

import javax.servlet.ServletException;
import javax.servlet.annotation.WebServlet;
import javax.servlet.http.HttpServlet;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import javax.sql.DataSource;
import java.io.IOException;
import java.sql.Connection;
import java.sql.PreparedStatement;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.util.ArrayList;
import java.util.List;
class Message{
    public String from;
    public String to;
    public String message;

    @Override
    public String toString() {
        return "Message{" +
                "from='" + from + '\'' +
                ", to='" + to + '\'' +
                ", message='" + message + '\'' +
                '}';
    }
}
@WebServlet("/message")
public class MessageWallServlet extends HttpServlet {
    //用来实现存档
    private ObjectMapper objectMapper = new ObjectMapper();
    //此时把消息保存到内存中，但是一旦重启服务器，内存数据就没了--->后续将其保存在数据库中使之更加科学
    //变成数据库之后，该list就可以删掉了
//    private List<Message> messageList = new ArrayList<>();
    @Override
    protected void doPost(HttpServletRequest req, HttpServletResponse resp) throws ServletException, IOException {
        //1.读取http请求中的body内容，将其中的json字符串转换成java对象---使用jackson库的ObjectMapper类中的readValue()方法，返回的是一个Message对象
        Message message = objectMapper.readValue(req.getInputStream(),Message.class);
        //2.将读取到的java对象保存在服务器中，这里新建一个list来存放Message对象
//        messageList.add(message);
        try {
            save(message);
        } catch (SQLException e) {
            e.printStackTrace();
        }
        //添加个日志，方便IDEA控制台可以看到
        System.out.println("message:" + message);//需要重写一下Message类的toString方法
        //3.返回响应,只要1返回一个200 ok 的响应就可以了，body可以没有
        //此时没有指定响应返回的数据类型，这里会有一个默认值为ContenType:text/plain表示纯文本
        resp.setStatus(200);//此时也可以不设置状态200，因为默认也是200的ok状态
        resp.getWriter().write("ok");//这个也可以不屑，因为页面不需要在成功存档后有任何的显示
    }

    private void save(Message message) throws SQLException {
        //通过jdbc往数据库中存数据
        //1.创建数据源
        DataSource dataSource = new MysqlDataSource();
        ((MysqlDataSource)dataSource).setUrl("jdbc:mysql://127.0.0.1:3306/message_wall?characterEncoding=utf8&useSSL=false");
        ((MysqlDataSource)dataSource).setUser("root");
        ((MysqlDataSource)dataSource).setPassword("123456");
        //2.建立连接
        Connection connection = dataSource.getConnection();
        //3.构造sql语句
        String sql = "insert into message values(?,?,?)";
        PreparedStatement statement = connection.prepareStatement(sql);
        statement.setString(1,message.from);
        statement.setString(2,message.to);
        statement.setString(3,message.message);
        //4.执行sql语句
        statement.executeUpdate();
        //5.释放资源，关闭连接
        statement.close();
        connection.close();
    }

    //用来实现读档
    @Override
    protected void doGet(HttpServletRequest req, HttpServletResponse resp) throws ServletException, IOException {
        //将内存中的Message对象组织成json字符串格式，返回到响应中
        //通过load方法将数据从数据库中读到List<Message>中
        List<Message> messageList = null;
        try {
            messageList = load();
        } catch (SQLException e) {
            e.printStackTrace();
        }
        String respJson = objectMapper.writeValueAsString(messageList);
        //将生成的字符串写到响应中去,同时需要告诉浏览器返回的响应body是一个json格式(utf8编码）
        resp.setContentType("application/json;charset=utf8");
        resp.getWriter().write(respJson);
    }

    private List<Message> load() throws SQLException {
        //通过jdbc从数据库中读取数据
        //1.创建数据源
        DataSource dataSource = new MysqlDataSource();
        ((MysqlDataSource)dataSource).setUrl("jdbc:mysql://127.0.0.1:3306/message_wall?characterEncoding=utf8&useSSL=false");
        ((MysqlDataSource)dataSource).setUser("root");
        ((MysqlDataSource)dataSource).setPassword("123456");
        //2.建立连接
        Connection connection = dataSource.getConnection();
        //3.构造sql语句
        String sql = "select * from message";
        //4.执行sql语句，返回的是一个ResultSet对象
        PreparedStatement statement = connection.prepareStatement(sql);
        ResultSet resultSet = statement.executeQuery();
        //5.遍历结果集合resultSet，并将其添加到List<Message>里面
        List<Message> messageList = new ArrayList<>();
        while (resultSet.next()){
            Message message = new Message();
            message.from = resultSet.getString("from");
            message.to = resultSet.getString("to");
            message.message = resultSet.getString("message");
            messageList.add(message);
        }
        //6.释放资源、关闭连接
        resultSet.close();
        statement.close();
        connection.close();
        return messageList;
    }
}
